/*
 * Copyright (C) 2019 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.android.tools.idea.npw.module.recipes.androidModule.src

import com.android.tools.idea.wizard.template.escapeKotlinIdentifier
import com.android.tools.idea.wizard.template.getMaterialComponentName
import com.android.tools.idea.wizard.template.renderIf

fun exampleInstrumentedTestKt(packageName: String, useAndroidX: Boolean, isLibraryProject: Boolean = false) = """
  package ${escapeKotlinIdentifier(packageName)}

  import ${getMaterialComponentName("android.support.test.InstrumentationRegistry", useAndroidX)}
  import ${getMaterialComponentName("android.support.test.runner.AndroidJUnit4", useAndroidX)}

  import org.junit.Test
  import org.junit.runner.RunWith

  import org.junit.Assert.*

  /**
   * Instrumented test, which will execute on an Android device.
   *
   * See [testing documentation](http://d.android.com/tools/testing).
   */
  @RunWith(AndroidJUnit4::class)
  class ExampleInstrumentedTest {
    @Test
    fun useAppContext() {
      // Context of the app under test.
      val appContext = InstrumentationRegistry.getInstrumentation().targetContext
      assertEquals("${packageName}${renderIf(isLibraryProject) { ".test" }}", appContext.packageName)
    }
  }
"""


fun exampleInstrumentedTestJava(packageName: String, useAndroidX: Boolean, isLibraryProject: Boolean = false) = """
  package ${packageName};

  import android.content.Context;
  import ${getMaterialComponentName("android.support.test.InstrumentationRegistry", useAndroidX)};
  import ${getMaterialComponentName("android.support.test.runner.AndroidJUnit4", useAndroidX)};

  import org.junit.Test;
  import org.junit.runner.RunWith;

  import static org.junit.Assert.*;

  /**
   * Instrumented test, which will execute on an Android device.
   *
   * @see <a href="http://d.android.com/tools/testing">Testing documentation</a>
   */
  @RunWith(AndroidJUnit4.class)
  public class ExampleInstrumentedTest {
      @Test
      public void useAppContext() {
          // Context of the app under test.
          Context appContext = InstrumentationRegistry.getInstrumentation().getTargetContext();
          assertEquals("${packageName}${renderIf(isLibraryProject) { ".test" }}", appContext.getPackageName());
      }
  }
"""
